package cn.reform.common.annotation;

import lombok.SneakyThrows;

import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;
import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.*;

/**
 * 枚举类型验证.<p>
 * 枚举类型验证注解，验证字段或参数值是否为指定枚举类中的有效值
 *
 * @author Jacky.sun
 * @date 2024/11/28
 */
@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = {ValidEnum.Validator.class})  // 指定验证器
@Documented
public @interface ValidEnum {

    String message() default "无效的枚举值";  // 默认错误消息

    Class<?>[] groups() default {};  // 分组校验，默认无分组

    Class<? extends Payload>[] payload() default {};  // 元数据，可以为空

    Class<? extends Enum<?>> enumClass();  // 要验证的枚举类

    boolean allowNull() default false;  // 是否允许为空

    class Validator implements ConstraintValidator<ValidEnum, Object> {
        private Class<? extends Enum<?>> enumClass;
        private boolean allowNull;

        @Override
        public void initialize(ValidEnum enumValue) {
            enumClass = enumValue.enumClass();
            allowNull = enumValue.allowNull();
        }

        @SneakyThrows
        @Override
        public boolean isValid(Object value, ConstraintValidatorContext constraintValidatorContext) {
            // 是否允许为null 或者 ""
            if (value == null || value.equals("")) {
                return allowNull;
            }

            // 参照枚举类为空说明不校验直接返回true
            if (enumClass == null) {
                return Boolean.TRUE;
            }

            //校验是否为枚举类中的值
            Enum<?>[] enumConstants = enumClass.getEnumConstants();
            for (Enum<?> enumConstant : enumConstants) {
                if (value.equals(enumConstant.toString())) {
                    return Boolean.TRUE;
                }
            }

            return Boolean.FALSE;
        }
    }
}